package com.ymttest.utils;

import java.io.*;
import java.util.Enumeration;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;
import java.util.zip.ZipOutputStream;

public class ZipUtils {

	public static void main(String args[]) throws Exception {
		zipFiles("resource", "1.ZIP");
	}

	/**
	 * @param path
	 */
	public static void unZip(String path,String savepath) {
		int count = -1;
		File file = null;
		InputStream is = null;
		FileOutputStream fos = null;
		BufferedOutputStream bos = null;
//		savepath = path.substring(0, path.lastIndexOf(".")) + File.separator; // 保存解压文件目录
		new File(savepath).mkdir(); // 创建保存目录
		ZipFile zipFile = null;
		try {
			zipFile = new ZipFile(path); 
			Enumeration<?> entries = zipFile.entries();
			while (entries.hasMoreElements()) {
				byte buf[] = new byte[1024];
				ZipEntry entry = (ZipEntry) entries.nextElement();
				String filename = entry.getName();
				boolean ismkdir = false;
				if (filename.lastIndexOf("/") != -1) { // 检查此文件是否带有文件夹
					ismkdir = true;
				}
				filename = savepath + filename;
				if (entry.isDirectory()) { // 如果是文件夹先创建
					file = new File(filename);
					file.mkdirs();
					continue;
				}
				file = new File(filename);
				if (!file.exists()) { // 如果是目录先创建
					if (ismkdir) {
						new File(filename.substring(0, filename.lastIndexOf("/"))).mkdirs(); // 目录先创建
					}
				}
				file.createNewFile(); // 创建文件
				is = zipFile.getInputStream(entry);
				fos = new FileOutputStream(file);
				bos = new BufferedOutputStream(fos, 1024);
				while ((count = is.read(buf)) > -1) {
					bos.write(buf, 0, count);
				}
				bos.flush();
				bos.close();
				fos.close();
				is.close();
			}
			zipFile.close();
		} catch (IOException ioe) {
			ioe.printStackTrace();
		} finally {
			try {
				if (bos != null) {
					bos.close();
				}
				if (fos != null) {
					fos.close();
				}
				if (is != null) {
					is.close();
				}
				if (zipFile != null) {
					zipFile.close();
				}
			} catch (Exception e) {
				e.printStackTrace();
			}
		}
	}

	/**
	 * 压缩某个目录或者文件
	 * 
	 * @param path    文件夹路径
	 * @param outFile
	 * @throws Exception
	 */
	public static void zipFiles(String path, String outFile) throws Exception {
		ZipOutputStream zos = new ZipOutputStream(new FileOutputStream(outFile));
		File sourcef = new File(path);
		Integer indexroot = sourcef.getAbsolutePath().length();
		if (sourcef.isFile()) {
			String entryName = sourcef.getName();
			ZipEntry entry = new ZipEntry(entryName);
			try {
				zos.putNextEntry(entry);
				InputStream is = new FileInputStream(path);
				int len = 0;
				while ((len = is.read()) != -1) {
					zos.write(len);
				}
				is.close();
			} catch (IOException e) {
				e.printStackTrace();
			}
			zos.close();
			return;
		} else if (sourcef.isDirectory()) {
			TestDataManager.getFilelist(path).forEach(filepath -> {
				String entryName = filepath.substring(indexroot + 1);
				ZipEntry entry = new ZipEntry(entryName);
				try {
					zos.putNextEntry(entry);
					InputStream is = new FileInputStream(filepath);
					int len = 0;
					while ((len = is.read()) != -1) {
						zos.write(len);
					}
					is.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
			});
			zos.close();
		} else {
			throw new Exception("file cannot zip:" + path);
		}
	}
}
